import os
import numpy as np
# file_path = '/home/lin/code/graduation_project/AD-Depth-Estimation/FeatDepth/splits/odom/train_files_all.txt'
# file_path = '/home/lin/code/graduation_project/AD-Depth-Estimation/FeatDepth/mono/datasets/splits/benchmark/eigen_to_benchmark_ids.npy'
# loadData = np.load(file_path)
# print(loadData)


# ----------------------------

file_path = '/home/lin/code/graduation_project/AD-Depth-Estimation/FeatDepth/splits/odom/train_files_all.txt'
save_path = '/home/lin/code/graduation_project/AD-Depth-Estimation/FeatDepth/splits/odom/train_files.txt'
f1 = open(save_path,'w')
with open(file_path, 'r') as f:
    text = f.readlines()
    for i in range(len(text)):
        if(text[i].split(' ')[0]=='1'):
            print(text[i].split(' '))
            f1.write(str(text[i].split(' ')[0])+' '+str(text[i].split(' ')[1])+' '+str(text[i].split(' ')[2]) )


